package com.dsprun.dspai.config;

import io.swagger.v3.oas.models.OpenAPI;
import io.swagger.v3.oas.models.info.Contact;
import io.swagger.v3.oas.models.info.Info;
import io.swagger.v3.oas.models.info.License;
import io.swagger.v3.oas.models.security.SecurityRequirement;
import io.swagger.v3.oas.models.security.SecurityScheme;
import org.springdoc.core.models.GroupedOpenApi;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;

@Configuration
public class OpenApiConfig {

    /**
     * 全局 OpenAPI 配置
     */
    @Bean
    public OpenAPI customOpenAPI() {
        return new OpenAPI()
                // 1. 添加API文档基本信息
                .info(new Info()
                        .title("DspAI人工智能套件 API 文档")
                        .version("1.0")
                        .description("基于SpringDoc OpenAPI3的接口文档")
                        .contact(new Contact()
                                .name("程易科技")
                                .email("dsp@dsprun.com")
                                .url("https://www.dsprun.com"))
                        .license(new License()
                                .name("Apache 2.0")
                                .url("https://www.apache.org/licenses/LICENSE-2.0")))
                // 2. 添加安全认证（如JWT）
                .addSecurityItem(new SecurityRequirement().addList("JWT"))
                .components(new io.swagger.v3.oas.models.Components()
                        .addSecuritySchemes("JWT", new SecurityScheme()
                                .name("JWT")
                                .type(SecurityScheme.Type.HTTP)
                                .scheme("bearer")
                                .bearerFormat("JWT")));
    }


    /**
     * 分组1：API（默认分组）
     */
    @Bean
    public GroupedOpenApi publicApi() {
        return GroupedOpenApi.builder()
                .group("public-api")
                .displayName("DSP应用")
                .pathsToMatch("/dsp/**")
                .pathsToMatch("/agent/**")
                .build();
    }

    /**
     * 分组2：公共API（默认分组）
     */
    @Bean
    public GroupedOpenApi AiApi() {
        return GroupedOpenApi.builder()
                .group("ai-api")
                .displayName("AI应用")
                .pathsToMatch("/api/ai/**")
                .build();
    }

    /**
     * 分组2：管理端API
     */
    @Bean
    public GroupedOpenApi adminApi() {
        return GroupedOpenApi.builder()
                .group("admin-api")
                .displayName("后台管理")
                .pathsToMatch("/api/admin/**")
                .addOpenApiCustomizer(openApi ->
                        openApi.info(new Info().title("管理端API").version("1.0")))
                .build();
    }
}